import axios, {AxiosInstance, AxiosRequestConfig, AxiosResponse} from 'axios'
import {ElMessage} from 'element-plus'
import router from '@/router'

interface ApiResponse<T = any> {
    code: number;
    msg: string;
    data: T;
}

// 定义自定义的 Axios 实例接口，明确其返回类型
interface CustomAxiosInstance extends AxiosInstance {
    <T = any>(config: AxiosRequestConfig): Promise<ApiResponse<T>>;

    <T = any>(url: string, config?: AxiosRequestConfig): Promise<ApiResponse<T>>;
}

const service: CustomAxiosInstance = axios.create({
    baseURL: '',
    timeout: 15000
}) as CustomAxiosInstance;

// 请求拦截器
service.interceptors.request.use(
    config => {
        // 从 localStorage 获取 token
        const token = localStorage.getItem('token')
        if (token) {
            // 添加 token 到请求头
            config.headers['satoken'] = token
        }
        return config
    },
    error => {
        return Promise.reject(error)
    }
)

// 响应拦截器
service.interceptors.response.use(
    (response: AxiosResponse) => {
        // 如果是二进制数据，直接返回
        if (response.config.responseType === 'blob') {
            return response;
        }

        const res = response.data
        if (res.code === 200) {
            return res
        } else if (res.code === 401) {
            // token 失效，清除本地 token 并跳转到登录页
            localStorage.removeItem('token')
            router.push('/login')
            ElMessage.error('登录已过期，请重新登录')
            return Promise.reject(new Error('登录已过期'))
        } else if (res.code === 403) {
            ElMessage.error('您没有权限，请联系管理员添加')
            return Promise.reject(new Error('您没有权限，请联系管理员添加'))
        } else {
            ElMessage.error(res.messsage || '请求失败')
            return Promise.reject(new Error(res.messsage || '请求失败'))
        }
    },
    error => {
        if (error.response && error.response.status === 401) {
            localStorage.removeItem('token')
            router.push('/login')
            ElMessage.error('登录已过期，请重新登录')
        } else {
            ElMessage.error(error.message || '请求失败')
        }
        return Promise.reject(error)
    }
)

export default service
